#include "stdafx.h"
#include "codecvt.h"

using namespace std;

ucs2_conversion::result
ucs2_conversion::do_in(mbstate_t&,
                       const char*  from, const char* from_end, const char*& from_next,
                       wchar_t* to, wchar_t* to_limit, wchar_t*& to_next) const
{
    size_t max_input = (from_end - from) & ~1;
    size_t max_output = (to_limit - to);
    size_t count = min(max_input/2, max_output);

    result res = ok;

    from_next = from;
    to_next = to;
    for (;count--; from_next += 2, ++to_next) {
        unsigned char c1 = *from_next, c2 = *(from_next + 1);
        *to_next = c1 | c2 << 8;
    }
    if (to_next == to && from_next == from_end - 1) res = partial;
    return res;
}

ucs2_conversion::result
ucs2_conversion::do_out(mbstate_t&,
                  const wchar_t* from, const wchar_t* from_end, const wchar_t*& from_next,
                  char* to, char* to_limit, char*& to_next) const
{
    size_t max_input = (from_end - from);
    size_t max_output = (to_limit - to) & ~1;
    size_t count = min(max_input, max_output/2);

    from_next = from;
    to_next = to;
    for (;count--; ++from_next, to_next += 2) {
        *(to_next + 0) = (char)(*from_next & 0xFF);
        *(to_next + 1) = (char)(*from_next >> 8 & 0xFF);
    }
    return ok;
}

typedef unsigned char uchar;

inline unsigned char
take_6_bits(unsigned value, size_t right_position)
{
    return uchar((value >> right_position) & 63);
}



inline size_t
most_signifant_bit_position(unsigned value)
{
    size_t result(0);

    size_t half = 16;
    for(; half > 0; half >>= 1) {
        if (1u << (result + half) <= value ) result += half;
    }
    return result + 1;
    //return *lower_bound(range(0u, 31u), \x -> (1 << x <= value));
}



utf8_conversion::result
utf8_conversion::do_in(mbstate_t&,
                       const char*  from, const char* from_end, const char*& from_next,
                       wchar_t* to, wchar_t* to_limit, wchar_t*& to_next) const
{
    from_next = from;
    to_next = to;

    for(; to_next < to_limit && from_next < from_end; ++to_next) {

        if (uchar(*from_next) < 0x80) *to_next = uchar(*from_next++);
        else {

            // 111zxxxx : z = 0 xxxx are data bits
            size_t zero_bit_pos = most_signifant_bit_position(~*from_next);
            size_t extra_bytes  = 7 - zero_bit_pos;

            if (size_t(from_end - from_next) < extra_bytes + 1)
                return partial;

            *to_next = uchar(*from_next++) & (wchar_t)((1 << (zero_bit_pos - 1)) - 1);

            for (;extra_bytes--; ++from_next) {
                *to_next = *to_next << 6  |  uchar(*from_next) & 63;
            }
        }
    }

    return ok;
}


utf8_conversion::result
utf8_conversion::do_out(mbstate_t&,
                  const wchar_t* from, const wchar_t* from_end, const wchar_t*& from_next,
                  char* to, char* to_limit, char*& to_next) const
{
        from_next = from;
        to_next = to;

        for (;from_next < from_end; ++from_next) {

            unsigned symbol = *from_next;

            if (symbol < 0x7F) {
                if (to_next < to_limit)
                    *to_next++ = (unsigned char)symbol;
                else
                    return ok;
            } else {

                size_t msb_pos = most_signifant_bit_position(symbol);
                size_t extra_bytes = msb_pos / 6;

                if (size_t(to_limit - to_next) >= extra_bytes + 1) {

                    *to_next = uchar(0xFF80 >> extra_bytes);
                    *to_next++ |= take_6_bits(symbol, extra_bytes*6);

                    for(;extra_bytes--;)
                        *to_next++ = 0x80 | take_6_bits(symbol, extra_bytes*6);
                }
                else
                    return ok;
            }
        }
        return ok;
}

